import os
import torch


class Config:
    root_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
    Bert_path = '/data0/jianyu10/PTM/huggingface_model_cache/chinese-roberta-wwm-ext'
    # Bert_path = 'D:/PTM/chinse-roberta-wwm-ext'
    bert_lr = 5e-5
    fc_lr = 0.01
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

    pooling = 'first-last-avg'
    train_path = root_path + '/data/traindata.csv'
    valid_path = root_path + '/data/devdata.csv'
    log_dir = root_path + '/Log/'
    batch_size = 20
    lr = 5e-5
    num_epochs = 300
    require_improvement = 60

    save_path = root_path + '/CheckPoint/'
